Pytorch划分數據集的方法

之前用過sklearn提供的划分數據集的函數，覺得超級方便。但是在使用TensorFlow和Pytorch的時候一直找不到類似的功能，之前搜索的關鍵字都是“pytorch split datas ...